before/after 效果的後端邏輯
在今天的文章中,我們將專注於實現 Before/After 效果 的後端邏輯。這個功能的核心是接收使用者上傳的圖片與提示詞,並透過 Google GenAI 的 API 生成新的圖片,最後將結果返回給前端。
功能需求
後端架構
後端使用 Flask 框架,並透過 Google GenAI 的 API 進行圖片生成。以下是主要的後端檔案結構:
utils/
before_after.py # 處理 API 請求的邏輯
photo_mix.py # 與 Google GenAI 的整合邏輯
config.py # 環境變數與設定
main.py # Flask 主程式
首先,我們需要設定 Google GenAI 的 API 金鑰,並將其存放在 .env
檔案中。
GEMINI_API_KEY=your_google_genai_api_key
在 config.py
中載入環境變數。
以下是 before_after.py
中的主要邏輯,負責處理圖片生成請求並返回結果:
在 before_after.py
中,我們還需要實現一個 API 路由,負責接收前端的請求,並調用 BeforeAfterService
的功能:
以下是 main.py
中處理 Before/After 效果 的主要程式碼,包含 API 路由與服務邏輯。